package com.dk;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

public class CodeGenerator {
    public static void main(String[] args) {

        /**
         * 创建代码生成器
         */
        AutoGenerator mpg = new AutoGenerator();

        /**
         * 全局配置
         */
        //创建全局配置
        GlobalConfig gc = new GlobalConfig();
        //获取当前项目
        String projectPath = System.getProperty("user.dir");
        //指定生成路径
        gc.setOutputDir(projectPath + "/springcloud_seckill_service_ssj/src/main/java");
        //设置作者
        gc.setAuthor("帅杰");
        //生成后是否打开资源管理器
        gc.setOpen(false);
        //生成后是否覆盖
        gc.setFileOverride(false);
        //设置主键策略    IdType.AUTO：主键自增
        gc.setIdType(IdType.AUTO);
        //自动生成的service接口首字母I去除
        gc.setServiceName("%sService");
        //设置日期类型
        gc.setDateType(DateType.ONLY_DATE);
        //实体属性是否生成 Swagger2(死哇死) 注解
//        gc.setSwagger2(true);
        //将全局配置放入代码生成器中
        mpg.setGlobalConfig(gc);

        /**
         * 数据源配置
         */
        //创建数据源
        DataSourceConfig dsc = new DataSourceConfig();
        //设置url
        dsc.setUrl("jdbc:mysql://127.0.0.1:3306/ershouche?useUnicode=true&useSSL=false&characterEncoding=utf8");
        //设置数据源驱动
        dsc.setDriverName("com.mysql.jdbc.Driver");
        //设置用户名
        dsc.setUsername("root");
        //设置密码
        dsc.setPassword("root");
        //设置数据库类型
        dsc.setDbType(DbType.MYSQL);
        //将数据源配置放入代码生成器中
        mpg.setDataSource(dsc);

        /**
         * 包配置
         */
        //创建包配置
        PackageConfig pc = new PackageConfig();
        //设置包名
        pc.setParent("com.dk");
        //设置实体类名称
        pc.setEntity("pojo");
        //设置控制层名称
        pc.setController("ontroller");
        //设置业务层名称
        pc.setService("service");
        //设置持久层名称
        pc.setMapper("mapper");
        //设置实现类名称
        pc.setServiceImpl("service.impl");
        //将包配置放入代码生成器
        mpg.setPackageInfo(pc);

        /**
         * 策略配置
         */
        //创建策略配置
        StrategyConfig strategy = new StrategyConfig();
        //指定生成的数据表名称    "tab_"+"\\w*" 表示：生成所有以 tab_ 开头的数据表
        strategy.setInclude("limited_special","car_message","car_brand","car_series");
        //设置数据库映射到实体的策略：大驼峰命名法
        strategy.setNaming(NamingStrategy.underline_to_camel);
        //不生成表的前缀：即去掉 tab_
//        strategy.setTablePrefix("tab_");
        //自动添加lombok注解 (老白可)
        strategy.setEntityLombokModel(true);
        /**
         * 逻辑删除
         */
        //指定逻辑删除字段
//        strategy.setLogicDeleteFieldName("is_delete");
        //去掉布尔值的 is_ 前缀
//        strategy.setEntityBooleanColumnRemoveIsPrefix(true);
        /**
         * RestFul风格
         */
        strategy.setRestControllerStyle(true);
        /**
         * url驼峰命名转换为 _
         */
        strategy.setControllerMappingHyphenStyle(true);
        //将策略配置放入代码生成器中
        mpg.setStrategy(strategy);

        /**
         * 执行
         */
        mpg.execute();
    }
}
